from __future__ import division
from functions import *
from utils import *
from gevi_classes import *

%matplotlib inline
%load_ext autoreload
%autoreload 2

%%javascript
IPython.OutputArea.auto_scroll_threshold = 1000;

# instantiate utility class
gr = Graph()

Load DATA

Data collected in 5 X 1min, with 1min pause in between 2921 images obtained in 60s

username = os.path.expanduser('~').split('/')[-1]
if username == "GP1514":
    print("At Imperial")
    mouseAPath = '/Volumes/DATA/DATA/Equalized Separation/2014 Oct 27/'
    mouseBPath = '/Volumes/DATA/DATA/Equalized Separation/2014 Oct 22/'
    mouseCPath = '/Volumes/DATA/DATA/Equalized Separation/2014 Oct 22/'
else:
    print("Using laptop")
    mouseAPath = '/Users/guillaume/Projects/GEVI-DATA/2014 Oct 27/'
    mouseBPath = '/Users/guillaume/Projects/GEVI-DATA/2014 Oct 22/'
    mouseCPath = '/Users/guillaume/Projects/GEVI-DATA/2014 Oct 28/'

At Imperial

discard = {
    'MouseA': 
    {
        3 : [0,1], 
        4 : [1,2], 
        5 : [2],
        6 : [2,3,4]
    }, 
    'MouseB' : 
    {
        3 : [2,4,5],
        4 : [1,2]
    }
}
# my choice
discard = {
    'MouseA': 
    {
        3 : [0,1,3], 
#         4 : [], 
#         5 : [],
        6 : [1]
    }, 
    'MouseB' : 
    {
        3 : [1,4],
#         4 : []
    },
    'MouseC' :
    {}
}

mouseA = Mouse('mouseA', mouseAPath, [3,4,5,6],discard['MouseA'] )
mouseB = Mouse('mouseB', mouseBPath, [2,3,4,5],discard['MouseB'] )
mouseC = Mouse('mouseC', mouseCPath, [4],discard['MouseC'] )

mouseA.loadData()
mouseB.loadData()
mouseC.loadData()

r = mouseA.experiments[0].repeats[1].ratio

# plotStats(mouseA,3, mode='var')
for k in mouseA.expNumbers:
# for k in [4]:
    plotStats(mouseA,k, mode='var')

for k in mouseA.expNumbers:
    plotStats(mouseA,k, mode='mean')

for k in mouseB.expNumbers:
    plotStats(mouseB,k, mode='var')

for k in mouseB.expNumbers:
    plotStats(mouseB,k, mode='mean')